import pandas as pd
from pandasai import Agent
from pandasai.llm.openai import OpenAI

# 读取 CSV 文件，使用原始字符串避免转义问题
df = pd.read_csv(r"D:\宁德时代股票.csv")
# 将 日期 列转换为日期时间格式
df['日期'] = pd.to_datetime(df['日期'])
# 设置 Date 列为索引
df.set_index('日期', inplace=True)

# 配置 Deepseek 模型
OpenAI._supported_chat_models.append('deepseek-chat')
llm_openai = OpenAI(
    api_base='https://api.deepseek.com/v1',
    api_token='sk-d57a50bd4d19428583093ee15d3e4510',  # 替换为实际的 API 令牌
    model='deepseek-chat'
)

# 创建 Agent
agent = Agent(dfs=[df], config={"llm": llm_openai})


# 找出数据中的缺失值
missing_value_prompt = "请找出数据框中每列的缺失值数量"
missing_value_response = agent.chat(missing_value_prompt)
print("\n每列缺失值数量信息：")
print(missing_value_response)


